package com.yu.util;

import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.net.URLDecoder;
import java.util.Enumeration;
import java.util.HashSet;
import java.util.Set;

public class ClassUtil {
    // 从包package中获取所有的Class
    public static Set<Class<?>> getClasses(String packageName) {
        Set<Class<?>> classSet = new HashSet<>();
        Enumeration<URL> dirs;
        try {
            String packageDirName = packageName.replace('.', '/'); // 起始扫描包
            dirs = Thread.currentThread().getContextClassLoader().getResources(packageDirName);
            while (dirs.hasMoreElements()) {
                String filePath = URLDecoder.decode(dirs.nextElement().getPath(), "utf-8"); // 包的物理路径
                findAndAddClassesInPackageByFile(packageName, filePath, classSet);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return classSet;
    }

    /**
     * 以文件的形式来获取包下的所有Class
     *
     * @param packageName
     * @param packagePath
     * @param classes
     */
    public static void findAndAddClassesInPackageByFile(String packageName, String packagePath, Set<Class<?>> classes) {
        File dir = new File(packagePath);
        if (!dir.exists() || !dir.isDirectory())
            return;

        // 循环所有文件
        for (File file : dir.listFiles()) {
            if (file.isDirectory()) { // 如果是目录 则继续扫描
                findAndAddClassesInPackageByFile(packageName + "." + file.getName(), file.getAbsolutePath(), classes);
            } else {
                // 如果是java类文件 去掉后面的.class 只留下类名
                String className = file.getName().substring(0, file.getName().length() - 6);
                try {
                    // 用Class.forName有会触发static方法，没有使用classLoader的load干净
                    classes.add(Thread.currentThread().getContextClassLoader().loadClass(packageName + '.' + className));
                } catch (ClassNotFoundException e) {
                    e.printStackTrace();
                }
            }
        }
    }

    // eg: UserService -> userService  str.length() >= 1
    public static String getBeanName(String str) {
        return Character.toLowerCase(str.charAt(0)) + str.substring(1);
    }
}

